package com.wz.leetcode.stackandqueue;

import java.util.Deque;
import java.util.LinkedList;

public class DailyTemperatures_739 {

    /**
     * 单调栈解法
     * @param temperatures
     * @return
     */
    public int[] dailyTemperatures(int[] temperatures) {
        int len = temperatures.length;
        int[] ans = new int[len];
        Deque<Integer> stack = new LinkedList<>();
        for (int i = 0; i < len; i++) {
            int temperature = temperatures[i];
            while (!stack.isEmpty() && temperature > temperatures[stack.peek()]) {
                int preIndex = stack.pop();
                ans[preIndex] = i - preIndex;
            }
            stack.push(i);
        }
        return ans;
    }

    /**
     * 逆序 + 暴力，利用已求解部分加速过程，消除重复求解过程。
     * @param temperatures
     * @return
     */
    public int[] dailyTemperatures2(int[] temperatures) {
        int len = temperatures.length;
        int[] ans = new int[len];
        for (int i = len - 2; i >= 0; i--) {
            // 利用已求解部分加速寻找第一个更大值
            for (int j = i + 1; j < len; j += ans[j]) {
                if (temperatures[j] > temperatures[i]) {
                    ans[i] = j - i;
                    break;
                } else if (ans[j] == 0) {
                    // 提前终止，后面没有更大的值
                    ans[i] = 0;
                    break;
                }
            }
        }
        return ans;
    }
}
